package com.zh.note.leetcode.backtracking;

import java.util.*;

/**
 * 39. 组合总和
 * 给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ，找出 candidates 中可以使数字和为目标数 target 的 所有 不同组合 ，并以列表形式返回。你可以按 任意顺序 返回这些组合。
 * <p>
 * candidates 中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同，则两种组合是不同的。
 * <p>
 * 对于给定的输入，保证和为 target 的不同组合数少于 150 个。
 */
public class LC_39_combinationSum {

    public List<List<Integer>> combinationSum(int[] candidates, int target) {
        List<List<Integer>> res = new ArrayList<>();
        Deque<Integer> path = new LinkedList<>();
        Arrays.sort(candidates);
        dfs(candidates, target, 0, 0, path, res);
        return res;

    }

    // public void dfs(int[] candidates, int target, int sum, int startIndex, Deque<Integer> path, List<List<Integer>> res) {
    //     if (sum > target) {
    //         return;
    //     }
    //     if (sum == target) {
    //         res.add(new ArrayList<>(path));
    //         return;
    //     }
    //     for (int i = startIndex; i < candidates.length; i++) {
    //         path.addLast(candidates[i]);
    //         sum += candidates[i];
    //         dfs(candidates, target, sum, i, path, res);// 不用i+1了，表示下一层可以重复读取当前的数
    //         sum -= candidates[i];
    //         path.removeLast();
    //     }
    // }
    // 剪枝
    public void dfs(int[] candidates, int target, int sum, int startIndex, Deque<Integer> path, List<List<Integer>> res) {
        if (sum > target) {
            return;
        }
        if (sum == target) {
            res.add(new ArrayList<>(path));
            return;
        }
        for (int i = startIndex; i < candidates.length; i++) {
            // 重点理解这里剪枝，前提是候选数组已经有序，
            if (target - candidates[i] < 0) {//剪枝
                break;
            }
            path.addLast(candidates[i]);
            sum += candidates[i];
            dfs(candidates, target, sum, i, path, res);// 不用i+1了，表示下一层可以重复读取当前的数
            sum -= candidates[i];
            path.removeLast();
        }
    }
}
